How to split json dataset and save it